package com.xh.chat.netty.handler;

import com.xh.chat.netty.dto.DataContentDTO;
import com.xh.chat.netty.event.EventHandlerFactory;
import com.xh.chat.netty.utils.ChannelUtil;
import com.xh.chat.utils.JsonUtils;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import lombok.extern.slf4j.Slf4j;

/**
 * 处理消息的handler
 * TextWebSocketFrame： 在netty中，是用于为websocket专门处理文本的对象，frame是消息的载体
 */
@Slf4j
public class ChatHandler extends SimpleChannelInboundHandler<TextWebSocketFrame> {

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) {
        // 1. 获取客户端发来的消息
        DataContentDTO dataContentDTO = JsonUtils.jsonToPojo(msg.text(), DataContentDTO.class);
        // 2. 判断消息类型，根据不同的类型来处理不同的业务
        EventHandlerFactory.getHandler(dataContentDTO.getAction()).handle(dataContentDTO, ctx.channel());
    }

    /**
     * 当客户端连接服务端之后（打开连接）
     * 获取客户端的channle，并且放到ChannelGroup中去进行管理
     */
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) {
        ChannelUtil.USER_CHANNEL_GROUP.add(ctx.channel());
    }

    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) {
        String channelId = ctx.channel().id().asShortText();
        log.info("客户端被移除，channelId为：" + channelId);

        // 当触发handlerRemoved，ChannelGroup会自动移除对应客户端的channel
        ChannelUtil.USER_CHANNEL_GROUP.remove(ctx.channel());
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
        cause.printStackTrace();
        // 发生异常之后关闭连接（关闭channel），随后从ChannelGroup中移除
        ctx.channel().close();
        ChannelUtil.USER_CHANNEL_GROUP.remove(ctx.channel());
    }
}
